import gym
from gym import Wrapper
import gym_textworld


def build_sb3_env(env_name):
    env_id = gym_textworld.make(env_name)
    env = gym.make(env_id)
    env = TextWorldSB3Env(env)

    return env


class TextWorldSB3Env(Wrapper):
    def __init__(self, env):
        super(TextWorldSB3Env, self).__init__(env)

    def _step(self, action):
        pass

    def _reset(self):
        pass

if __name__ == "__main__":
    env = build_sb3_env('twcc_easy_level5_gamesize100_step50_seed9_train')
    env.reset()